#!/bin/bash

# 作业配置
#SBATCH --job-name=train-segment-segformer # 作业名称
#SBATCH --output=segformer_output_%j.log # 标准输出日志（%j 会被替换为作业ID）
#SBATCH --error=segformer_error_%j.log # 标准错误日志

# 资源配置
#SBATCH --nodes=1 # 申请 1 个节点
#SBATCH --ntasks=1 # 总共 1 个任务（MPI 进程数）
#SBATCH --cpus-per-task=48 # 每个任务使用 48 个 CPU 核
#SBATCH --partition=gpu-a10 # 指定分区
#SBATCH --gres=gpu:1 # 指定显卡数量

# 跳转目录
cd ~/workspace/segment

# 激活虚拟环境
source ~/.bashrc
conda activate seg

# 启动作业
python train.py --config config/segformer.py